package ai.mspbots.poc.help;

import org.deeplearning4j.models.embeddings.wordvectors.WordVectors;

public class WordVectorsThreadLocal {

    private static final ThreadLocal<WordVectors> WORD_VECTORS = new ThreadLocal<>();

    public static void setWordVectors(WordVectors wordVectors) {
        WORD_VECTORS.set(wordVectors);
    }

    public static WordVectors getWordVectors() {
        return WORD_VECTORS.get();
    }

    public static void remove() {
        WORD_VECTORS.remove();
    }
}
